ALRA: Adaptive Local Relational Alignment for Logit-Based Pre-training Distillation of Autoregressive Language Models
自适应局部关系对齐:自回归语言模型预训练蒸馏的 logit 方法
🧭 知识蒸馏训练优化
VJ Technologies · AJ Technologies · Vietnam National University · University of Information Technology
ALRA 结合学生提议与教师锚点,自适应局部对齐改进 logit 蒸馏,提升预训练学生模型零样本性能
📌 工作参考价值(4 点)
- 在生成式推荐(如 Semantic ID、商品标题生成)的蒸馏中,可借鉴 ALRA 的候选 token 选择策略:让学生提出高概率 token,同时将教师的 top1 token 作为锚点加入,避免只用教师或学生导致候选集偏差,尤其适合训练早期学生排序不准确的场景。
- Adaptive Local Divergence 用单位系数替代教师质量系数,防止低概率但语义重要的 token 区域被忽略;这可以迁移到电商长尾商品或 query 的蒸馏,保护稀有但相关的 token 信号不被全局分布淹没。
- Student-Weighted Pairwise Relational Alignment 强调学生概率差小但教师偏好的 token 对,可用于校准推荐生成中 top-k 的局部排序,提升生成结果的多样性与准确性。
- 位置特定且 batch 自适应的 token 数量选择,减少了手工设定 top-k 的超参负担,适合在线学习或增量更新场景,能动态适应数据分布变化。